/*
 * @lc app=leetcode.cn id=1305 lang=cpp
 * @lcpr version=30204
 *
 * [1305] 两棵二叉搜索树中的所有元素
 */

// @lcpr-template-start
using namespace std;
#include <algorithm>
#include <array>
#include <bitset>
#include <climits>
#include <deque>
#include <functional>
#include <iostream>
#include <list>
#include <queue>
#include <stack>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
// @lcpr-template-end

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};

// @lc code=start

class Solution
{
public:
    vector<int> getAllElements(TreeNode *root1, TreeNode *root2)
    {
        vector<int> v1;
        zhongxu(root1, v1);
        vector<int> v2;
        zhongxu(root2, v2);
        return chapai(v1, v2);
    }

    void zhongxu(TreeNode *root, vector<int> &v)
    {
        if (root == nullptr)
            return;
        zhongxu(root->left, v);
        v.push_back(root->val);
        zhongxu(root->right, v);
    }

    vector<int> chapai(const vector<int> &v1, const vector<int> &v2)
    {
        vector<int> res;
        auto it1 = v1.begin(), it2 = v2.begin();
        while (it1 != v1.end() && it2 != v2.end())
        {
            if (*it1 < *it2)
            {
                res.push_back(*it1);
                it1++;
            }
            else
            {
                res.push_back(*it2);
                it2++;
            }
        }
        res.insert(res.end(), it1, v1.end());
        res.insert(res.end(), it2, v2.end());
        return res;
    }
};
// @lc code=end

/*
// @lcpr case=start
// [2,1,4]\n[1,0,3]\n
// @lcpr case=end

// @lcpr case=start
// [1,null,8]\n[8,1]\n
// @lcpr case=end

 */
